<?php

namespace Service\Database\Drivers;

use Service\Database\Driver;

/**
 * Mongo数据库驱动
 */
class Mongo extends Driver {
	
	protected $_mongo           =   null; // MongoDb Object
	protected $_collection      =   null; // MongoCollection Object
	protected $_dbName          =   ''; // dbName
	protected $_collectionName  =   ''; // collectionName
	protected $_cursor          =   null; // MongoCursor Object
	protected $comparison       =   array('neq'=>'ne','ne'=>'ne','gt'=>'gt','egt'=>'gte','gte'=>'gte','lt'=>'lt','elt'=>'lte','lte'=>'lte','in'=>'in','not in'=>'nin','nin'=>'nin');
	
	/**
	 * 架构函数 读取数据库配置信息
	 * @access public
	 * @param array $config 数据库配置数组
	 */
	public function __construct($config=''){
		if ( !class_exists('mongoClient') ) {
			E(L('_NOT_SUPPORT_').':Mongo');
		}
		if(!empty($config)) {
			$this->config           =   array_merge($this->config,$config);
			if(empty($this->config['params'])){
				$this->config['params'] =   array();
			}
		}
	}
	
	/**
	 * 连接数据库方法
	 * @access public
	 */
	public function connect($config='',$linkNum=0) {
		if ( !isset($this->linkID[$linkNum]) ) {
			if(empty($config))  $config =   $this->config;
			$host = 'mongodb://'.($config['username']?"{$config['username']}":'').($config['password']?":{$config['password']}@":'').$config['hostname'].($config['hostport']?":{$config['hostport']}":'').'/'.($config['database']?"{$config['database']}":'');
			try{
				$this->linkID[$linkNum] = new \mongoClient( $host,$this->config['params']);
			}catch (\MongoConnectionException $e){
				E($e->getmessage());
			}
		}
		return $this->linkID[$linkNum];
	}
	
	/**
	 * 切换当前操作的Db和Collection
	 * @access public
	 * @param string $collection  collection
	 * @param string $db  db
	 * @param boolean $master 是否主服务器
	 * @return void
	 */
	public function switchCollection($collection,$db='',$master=true){
		// 当前没有连接 则首先进行数据库连接
		if ( !$this->_linkID ) $this->initConnect($master);
		try{
			if(!empty($db)) { // 传人Db则切换数据库
				// 当前MongoDb对象
				$this->_dbName  =  $db;
				$this->_mongo = $this->_linkID->selectDb($db);
			}
			// 当前MongoCollection对象
			if($this->config['debug']) {
				$this->queryStr   =  $this->_dbName.'.getCollection('.$collection.')';
			}
			if($this->_collectionName != $collection) {
				$this->queryTimes++;
				$this->debug(true);
				$this->_collection =  $this->_mongo->selectCollection($collection);
				$this->debug(false);
				$this->_collectionName  = $collection; // 记录当前Collection名称
			}
		}catch (\MongoException $e){
			E($e->getMessage());
		}
	}
	
	/**
	 * 释放查询结果
	 * @access public
	 */
	public function free() {
		$this->_cursor = null;
	}
	
	/**
	 * 执行命令
	 * @access public
	 * @param array $command  指令
	 * @return array
	 */
	public function command($command=array(), $options=array()) {
		$cache  =  isset($options['cache'])?$options['cache']:false;
		if($cache) { // 查询缓存检测
			$key =  is_string($cache['key'])?$cache['key']:md5(serialize($command));
			$value   =  S($key,'','',$cache['type']);
			if(false !== $value) {
				return $value;
			}
		}
		$this->executeTimes++;
		try{
			if($this->config['debug']) {
				$this->queryStr   =  $this->_dbName.'.'.$this->_collectionName.'.runCommand(';
				$this->queryStr  .=  json_encode($command);
				$this->queryStr  .=  ')';
			}
			$this->debug(true);
			$result   = $this->_mongo->command($command);
			$this->debug(false);
			
			if($cache && $result['ok']) { // 查询缓存写入
				S($key,$result,$cache['expire'],$cache['type']);
			}
			return $result;
		} catch (\MongoCursorException $e) {
			E($e->getMessage());
		}
	}
	
	/**
	 * 执行语句
	 * @access public
	 * @param string $code  sql指令
	 * @param array $args  参数
	 * @return mixed
	 */
	public function execute($code,$args=array()) {
		$this->executeTimes++;
		$this->debug(true);
		$this->queryStr = 'execute:'.$code;
		$result   = $this->_mongo->execute($code,$args);
		$this->debug(false);
		if($result['ok']) {
			return $result['retval'];
		}else{
			E($result['errmsg']);
		}
	}
	
	/**
	 * 关闭数据库
	 * @access public
	 */
	public function close() {
		if($this->_linkID) {
			$this->_linkID->close();
			$this->_linkID = null;
			$this->_mongo = null;
			$this->_collection =  null;
			$this->_cursor = null;
		}
	}
	
	/**
	 * 数据库错误信息
	 * @access public
	 * @return string
	 */
	public function error() {
		$this->error = $this->_mongo->lastError();
		trace($this->error,'','ERR');
		return $this->error;
	}
	
	/**
	 * 插入记录
	 * @access public
	 * @param mixed $data 数据
	 * @param array $options 参数表达式
	 * @param boolean $replace 是否replace
	 * @return false | integer
	 */
	public function insert($data,$options=array(),$replace=false) {
		if(isset($options['table'])) {
			$this->switchCollection($options['table']);
		}
		$this->model  =   $options['model'];
		$this->executeTimes++;
		if($this->config['debug']) {
			$this->queryStr   =  $this->_dbName.'.'.$this->_collectionName.'.insert(';
			$this->queryStr   .= $data?json_encode($data):'{}';
			$this->queryStr   .= ')';
		}
		try{
			$this->debug(true);
			$result =  $replace?   $this->_collection->save($data):  $this->_collection->insert($data);
			$this->debug(false);
			if($result) {
				$_id    = $data['_id'];
				if(is_object($_id)) {
					$_id = $_id->__toString();
				}
				$this->lastInsID    = $_id;
			}
			return $result;
		} catch (\MongoCursorException $e) {
			E($e->getMessage());
		}
	}
	
	/**
	 * 插入多条记录
	 * @access public
	 * @param array $dataList 数据
	 * @param array $options 参数表达式
	 * @return bool
	 */
	public function insertAll($dataList,$options=array()) {
		if(isset($options['table'])) {
			$this->switchCollection($options['table']);
		}
		$this->model  =   $options['model'];
		$this->executeTimes++;
		try{
			$this->debug(true);
			$result =  $this->_collection->batchInsert($dataList);
			$this->debug(false);
			return $result;
		} catch (\MongoCursorException $e) {
			E($e->getMessage());
		}
	}
	
	/**
	 * 生成下一条记录ID 用于自增非MongoId主键
	 * @access public
	 * @param string $pk 主键名
	 * @return integer
	 */
	public function getMongoNextId($pk) {
		if($this->config['debug']) {
			$this->queryStr   =  $this->_dbName.'.'.$this->_collectionName.'.find({},{'.$pk.':1}).sort({'.$pk.':-1}).limit(1)';
		}
		try{
			$this->debug(true);
			$result   =  $this->_collection->find(array(),array($pk=>1))->sort(array($pk=>-1))->limit(1);
			$this->debug(false);
		} catch (\MongoCursorException $e) {
			E($e->getMessage());
		}
		$data = $result->getNext();
		return isset($data[$pk])?$data[$pk]+1:1;
	}
	
	/**
	 * 更新记录
	 * @access public
	 * @param mixed $data 数据
	 * @param array $options 表达式
	 * @return bool
	 */
	public function update($data,$options) {
		if(isset($options['table'])) {
			$this->switchCollection($options['table']);
		}
		$this->executeTimes++;
		$this->model  =   $options['model'];
		$query   = $this->parseWhere(isset($options['where'])?$options['where']:array());
		$set  =  $this->parseSet($data);
		if($this->config['debug']) {
			$this->queryStr   =  $this->_dbName.'.'.$this->_collectionName.'.update(';
			$this->queryStr   .= $query?json_encode($query):'{}';
			$this->queryStr   .=  ','.json_encode($set).')';
		}
		try{
			$this->debug(true);
			if(isset($options['limit']) && $options['limit'] == 1) {
				$multiple   =   array("multiple" => false);
			}else{
				$multiple   =   array("multiple" => true);
			}
			$result   = $this->_collection->update($query,$set,$multiple);
			$this->debug(false);
			return $result;
		} catch (\MongoCursorException $e) {
			E($e->getMessage());
		}
	}
	
	/**
	 * 删除记录
	 * @access public
	 * @param array $options 表达式
	 * @return false | integer
	 */
	public function delete($options=array()) {
		if(isset($options['table'])) {
			$this->switchCollection($options['table']);
		}
		$query   = $this->parseWhere(isset($options['where'])?$options['where']:array());
		$this->model  =   $options['model'];
		$this->executeTimes++;
		if($this->config['debug']) {
			$this->queryStr   =  $this->_dbName.'.'.$this->_collectionName.'.remove('.json_encode($query).')';
		}
		try{
			$this->debug(true);
			$result   = $this->_collection->remove($query);
			$this->debug(false);
			return $result;
		} catch (\MongoCursorException $e) {
			E($e->getMessage());
		}
	}
	
	/**
	 * 清空记录
	 * @access public
	 * @param array $options 表达式
	 * @return false | integer
	 */
	public function clear($options=array()){
		if(isset($options['table'])) {
			$this->switchCollection($options['table']);
		}
		$this->model  =   $options['model'];
		$this->executeTimes++;
		if($this->config['debug']) {
			$this->queryStr   =  $this->_dbName.'.'.$this->_collectionName.'.remove({})';
		}
		try{
			$this->debug(true);
			$result   =  $this->_collection->drop();
			$this->debug(false);
			return $result;
		} catch (\MongoCursorException $e) {
			E($e->getMessage());
		}
	}
	
	/**
	 * 查找记录
	 * @access public
	 * @param array $options 表达式
	 * @return mixed
	 */
	public function select($options=array()) {
		if(isset($options['table'])) {
			$this->switchCollection($options['table'],'',false);
		}
		$this->model  =   $options['model'];
		$this->queryTimes++;
		$query  =  $this->parseWhere(isset($options['where'])?$options['where']:array());
		$field  =  $this->parseField(isset($options['field'])?$options['field']:array());
		try{
			if($this->config['debug']) {
				$this->queryStr   =  $this->_dbName.'.'.$this->_collectionName.'.find(';
				$this->queryStr  .=  $query? json_encode($query):'{}';
				if(is_array($field) && count($field)) {
					foreach ($field as $f=>$v)
						$_field_array[$f] = $v ? 1 : 0;
						
						$this->queryStr  .=  $field? ', '.json_encode($_field_array):', {}';
				}
				$this->queryStr  .=  ')';
			}
			$this->debug(true);
			$_cursor   = $this->_collection->find($query,$field);
			if(!empty($options['order'])) {
				$order   =  $this->parseOrder($options['order']);
				if($this->config['debug']) {
					$this->queryStr .= '.sort('.json_encode($order).')';
				}
				$_cursor =  $_cursor->sort($order);
			}
			if(isset($options['page'])) { // 根据页数计算limit
				list($page,$length)   =   $options['page'];
				$page    =  $page>0 ? $page : 1;
				$length  =  $length>0 ? $length : (is_numeric($options['limit'])?$options['limit']:20);
				$offset  =  $length*((int)$page-1);
				$options['limit'] =  $offset.','.$length;
			}
			if(isset($options['limit'])) {
				list($offset,$length) =  $this->parseLimit($options['limit']);
				if(!empty($offset)) {
					if($this->config['debug']) {
						$this->queryStr .= '.skip('.intval($offset).')';
					}
					$_cursor =  $_cursor->skip(intval($offset));
				}
				if($this->config['debug']) {
					$this->queryStr .= '.limit('.intval($length).')';
				}
				$_cursor =  $_cursor->limit(intval($length));
			}
			$this->debug(false);
			$this->_cursor =  $_cursor;
			$resultSet  =  iterator_to_array($_cursor);
			return $resultSet;
		} catch (\MongoCursorException $e) {
			E($e->getMessage());
		}
	}
	
	/**
	 * 查找某个记录
	 * @access public
	 * @param array $options 表达式
	 * @return array
	 */
	public function find($options=array()){
		$options['limit'] = 1;
		$find = $this->select($options);
		return array_shift($find);
	}
	
	/**
	 * 统计记录数
	 * @access public
	 * @param array $options 表达式
	 * @return 
	 */
	public function count($options=array()){
		if(isset($options['table'])) {
			$this->switchCollection($options['table'],'',false);
		}
		$this->model  =   $options['model'];
		$this->queryTimes++;
		$query  =  $this->parseWhere(isset($options['where'])?$options['where']:array());
		if($this->config['debug']) {
			$this->queryStr   =  $this->_dbName.'.'.$this->_collectionName;
			$this->queryStr   .= $query?'.find('.json_encode($query).')':'';
			$this->queryStr   .= '.count()';
		}
		try{
			$this->debug(true);
			$count   = $this->_collection->count($query);
			$this->debug(false);
			return $count;
		} catch (\MongoCursorException $e) {
			E($e->getMessage());
		}
	}
	
	public function group($keys,$initial,$reduce,$options=array()){
		if(isset($options['table']) && $this->_collectionName != $options['table']) {
			$this->switchCollection($options['table'],'',false);
		}
		
		$cache  =  isset($options['cache'])?$options['cache']:false;
		if($cache) {
			$key    =  is_string($cache['key'])?$cache['key']:md5(serialize($options));
			$value  =  S($key,'','',$cache['type']);
			if(false !== $value) {
				return $value;
			}
		}
		
		$this->model  =   $options['model'];
		$this->queryTimes++;
		$query  =  $this->parseWhere(isset($options['where'])?$options['where']:array());
		
		if($this->config['debug']) {
			$this->queryStr   =  $this->_dbName.'.'.$this->_collectionName.'.group({key:'.json_encode($keys).',cond:'.
					json_encode($options['condition']) . ',reduce:' .
					json_encode($reduce).',initial:'.
					json_encode($initial).'})';
		}
		try{
			$this->debug(true);
			$option = array('condition'=>$options['condition'], 'finalize'=>$options['finalize'], 'maxTimeMS'=>$options['maxTimeMS']);
			$group = $this->_collection->group($keys,$initial,$reduce,$options);
			$this->debug(false);
			
			if($cache && $group['ok'])
				S($key,$group,$cache['expire'],$cache['type']);
				
				return $group;
		} catch (\MongoCursorException $e) {
			E($e->getMessage());
		}
	}
	
	/**
	 * 取得数据表的字段信息
	 * @access public
	 * @return array
	 */
	public function getFields($collection=''){
		if(!empty($collection) && $collection != $this->_collectionName) {
			$this->switchCollection($collection,'',false);
		}
		$this->queryTimes++;
		if($this->config['debug']) {
			$this->queryStr   =  $this->_dbName.'.'.$this->_collectionName.'.findOne()';
		}
		try{
			$this->debug(true);
			$result   =  $this->_collection->findOne();
			$this->debug(false);
		} catch (\MongoCursorException $e) {
			E($e->getMessage());
		}
		if($result) { // 存在数据则分析字段
			$info =  array();
			foreach ($result as $key=>$val){
				$info[$key] =  array(
						'name'  =>  $key,
						'type'  =>  getType($val),
				);
			}
			return $info;
		}
		// 暂时没有数据 返回false
		return false;
	}
	
	/**
	 * 取得当前数据库的collection信息
	 * @access public
	 */
	public function getTables(){
		if($this->config['debug']) {
			$this->queryStr   =  $this->_dbName.'.getCollenctionNames()';
		}
		$this->queryTimes++;
		$this->debug(true);
		$list   = $this->_mongo->listCollections();
		$this->debug(false);
		$info =  array();
		foreach ($list as $collection){
			$info[]   =  $collection->getName();
		}
		return $info;
	}
	
	/**
	 * 取得当前数据库的对象
	 * @access public
	 * @return object mongoClient
	 */
	public function getDB(){
		return $this->_mongo;
	}
	
	/**
	 * 取得当前集合的对象
	 * @access public
	 * @return object MongoCollection
	 */
	public function getCollection(){
		return $this->_collection;
	}
	
	/**
	 * set分析
	 * @access protected
	 * @param array $data
	 * @return string
	 */
	protected function parseSet($data) {
		$result   =  array();
		foreach ($data as $key=>$val){
			if(is_array($val)) {
				switch($val[0]) {
					case 'inc':
						$result['$inc'][$key]  =  (int)$val[1];
						break;
					case 'set':
					case 'unset':
					case 'push':
					case 'pushall':
					case 'addtoset':
					case 'pop':
					case 'pull':
					case 'pullall':
						$result['$'.$val[0]][$key] = $val[1];
						break;
					default:
						$result['$set'][$key] =  $val;
				}
			}else{
				$result['$set'][$key]    = $val;
			}
		}
		return $result;
	}
	
	/**
	 * order分析
	 * @access protected
	 * @param mixed $order
	 * @return array
	 */
	protected function parseOrder($order) {
		if(is_string($order)) {
			$array   =  explode(',',$order);
			$order   =  array();
			foreach ($array as $key=>$val){
				$arr  =  explode(' ',trim($val));
				if(isset($arr[1])) {
					$arr[1]  =  $arr[1]=='asc'?1:-1;
				}else{
					$arr[1]  =  1;
				}
				$order[$arr[0]]    = $arr[1];
			}
		}
		return $order;
	}
	
	/**
	 * limit分析
	 * @access protected
	 * @param mixed $limit
	 * @return array
	 */
	protected function parseLimit($limit) {
		if(strpos($limit,',')) {
			$array  =  explode(',',$limit);
		}else{
			$array   =  array(0,$limit);
		}
		return $array;
	}
	
	/**
	 * field分析
	 * @access protected
	 * @param mixed $fields
	 * @return array
	 */
	public function parseField($fields){
		if(empty($fields)) {
			$fields    = array();
		}
		if(is_string($fields)) {
			$_fields    = explode(',',$fields);
			$fields     = array();
			foreach ($_fields as $f)
				$fields[$f] = true;
		}elseif(is_array($fields)) {
			$_fields    = $fields;
			$fields     = array();
			foreach ($_fields as $f=>$v) {
				if(is_numeric($f))
					$fields[$v] = true;
					else
						$fields[$f] = $v ? true : false;
			}
		}
		return $fields;
	}
	
	/**
	 * where分析
	 * @access protected
	 * @param mixed $where
	 * @return array
	 */
	public function parseWhere($where){
		$query   = array();
		$return     = array();
		$_logic     = '$and';
		if(isset($where['_logic'])){
			$where['_logic']    = strtolower($where['_logic']);
			$_logic             = in_array($where['_logic'], array('or','xor','nor', 'and'))?'$'.$where['_logic']:$_logic;
			unset($where['_logic']);
		}
		foreach ($where as $key=>$val){
			if('_id' != $key && 0===strpos($key,'_')) {
				// 解析特殊条件表达式
				$parse   = $this->parseThinkWhere($key,$val);
				$query   = array_merge($query,$parse);
			}else{
				// 查询字段的安全过滤
				if(!preg_match('/^[A-Z_\|\&\-.a-z0-9]+$/',trim($key))){
					E(L('_ERROR_QUERY_').':'.$key);
				}
				$key = trim($key);
				if(strpos($key,'|')) {
					$array   =  explode('|',$key);
					$str   = array();
					foreach ($array as $k){
						$str[]   = $this->parseWhereItem($k,$val);
					}
					$query['$or'] =    $str;
				}elseif(strpos($key,'&')){
					$array   =  explode('&',$key);
					$str   = array();
					foreach ($array as $k){
						$str[]   = $this->parseWhereItem($k,$val);
					}
					$query   = array_merge($query,$str);
				}else{
					$str   = $this->parseWhereItem($key,$val);
					$query   = array_merge($query,$str);
				}
			}
		}
		if($_logic == '$and')
			return $query;
			
			foreach($query as $key=>$val)
				$return[$_logic][]  = array($key=>$val);
				
				return $return;
	}
	
	/**
	 * 特殊条件分析
	 * @access protected
	 * @param string $key
	 * @param mixed $val
	 * @return string
	 */
	protected function parseThinkWhere($key,$val) {
		$query   = array();
		$_logic = array('or','xor','nor', 'and');
		
		switch($key) {
			case '_query': // 字符串模式查询条件
				parse_str($val,$query);
				if(isset($query['_logic']) && strtolower($query['_logic']) == 'or' ) {
					unset($query['_logic']);
					$query['$or']   =  $query;
				}
				break;
			case '_complex': // 子查询模式查询条件
				$__logic = strtolower($val['_logic']);
				if(isset($val['_logic']) && in_array($__logic, $_logic) ) {
					unset($val['_logic']);
					$query['$'.$__logic]   =  $val;
				}
				break;
			case '_string':// MongoCode查询
				$query['$where']  = new \MongoCode($val);
				break;
		}
		//兼容 MongoClient OR条件查询方法
		if(isset($query['$or']) && !is_array(current($query['$or']))) {
			$val = array();
			foreach ($query['$or'] as $k=>$v)
				$val[] = array($k=>$v);
				$query['$or'] = $val;
		}
		return $query;
	}
	
	/**
	 * where子单元分析
	 * @access protected
	 * @param string $key
	 * @param mixed $val
	 * @return array
	 */
	protected function parseWhereItem($key,$val) {
		$query   = array();
		if(is_array($val)) {
			if(is_string($val[0])) {
				$con  =  strtolower($val[0]);
				if(in_array($con,array('neq','ne','gt','egt','gte','lt','lte','elt'))) { // 比较运算
					$k = '$'.$this->comparison[$con];
					$query[$key]  =  array($k=>$val[1]);
				}elseif('like'== $con){ // 模糊查询 采用正则方式
					$query[$key]  =  new \MongoRegex("/".$val[1]."/");
				}elseif('mod'==$con){ // mod 查询
					$query[$key]   =  array('$mod'=>$val[1]);
				}elseif('regex'==$con){ // 正则查询
					$query[$key]  =  new \MongoRegex($val[1]);
				}elseif(in_array($con,array('in','nin','not in'))){ // IN NIN 运算
					$data = is_string($val[1])? explode(',',$val[1]):$val[1];
					$k = '$'.$this->comparison[$con];
					$query[$key]  =  array($k=>$data);
				}elseif('all'==$con){ // 满足所有指定条件
					$data = is_string($val[1])? explode(',',$val[1]):$val[1];
					$query[$key]  =  array('$all'=>$data);
				}elseif('between'==$con){ // BETWEEN运算
					$data = is_string($val[1])? explode(',',$val[1]):$val[1];
					$query[$key]  =  array('$gte'=>$data[0],'$lte'=>$data[1]);
				}elseif('not between'==$con){
					$data = is_string($val[1])? explode(',',$val[1]):$val[1];
					$query[$key]  =  array('$lt'=>$data[0],'$gt'=>$data[1]);
				}elseif('exp'==$con){ // 表达式查询
					$query['$where']  = new \MongoCode($val[1]);
				}elseif('exists'==$con){ // 字段是否存在
					$query[$key]  = array('$exists'=>(bool)$val[1]);
				}elseif('size'==$con){ // 限制属性大小
					$query[$key]  = array('$size'=>intval($val[1]));
				}elseif('type'==$con){ // 限制字段类型 1 浮点型 2 字符型 3 对象或者MongoDBRef 5 MongoBinData 7 MongoId 8 布尔型 9 MongoDate 10 NULL 15 MongoCode 16 32位整型 17 MongoTimestamp 18 MongoInt64 如果是数组的话判断元素的类型
					$query[$key]  = array('$type'=>intval($val[1]));
				}else{
					$query[$key]  =  $val;
				}
				return $query;
			}
		}
		$query[$key]  =  $val;
		return $query;
	}
}
